#ifndef __SPLITTOOL_H__
#define __SPLITTOOL_H__

#include "../cppjieba/Jieba.hpp"

#include <vector>
#include <string>

using std::string;
using std::vector;

const char* const DICT_PATH = "../cppjieba/dict/jieba.dict.utf8";
const char* const HMM_PATH = "../cppjieba/dict/hmm_model.utf8";
const char* const USER_DICT_PATH = "../cppjieba/dict/user.dict.utf8";
const char* const IDF_PATH = "../cppjieba/dict/idf.utf8";
const char* const STOP_WORD_PATH = "../cppjieba/dict/stop_words.utf8";

// const string DICT_PATH = "../cppjieba/dict/jieba.dict.utf8";
// const string HMM_PATH = "../cppjieba/dict/hmm_model.utf8";
// const string USER_DICT_PATH = "../cppjieba/dict/user.dict.utf8";
// const string IDF_PATH = "../cppjieba/dict/idf.utf8";
// const string STOP_WORD_PATH = "../cppjieba/dict/stop_words.utf8";
class SplitTool{
public:
    SplitTool()
    : _jieba(DICT_PATH, HMM_PATH, USER_DICT_PATH,IDF_PATH,STOP_WORD_PATH)
    {}

    ~SplitTool() {

    }

    vector<string> cut(const string &lines) {
        vector<string> words;
        _jieba.CutAll(lines, words);
        return words;
    }
    
private:
    cppjieba::Jieba _jieba;
};

#endif
